Main

In your final repo, there should be an R markdown file that organizes all computational steps for evaluating your proposed Facial Expression Recognition framework.

This file is currently a template for running evaluation experiments. You should update it according to your codes but following precisely the same structure.

if(!require("EBImage")){
  install.packages("BiocManager")
  BiocManager::install("EBImage")
}
if(!require("R.matlab")){
  install.packages("R.matlab")
}
if(!require("readxl")){
  install.packages("readxl")
}

if(!require("dplyr")){
  install.packages("dplyr")
}
if(!require("readxl")){
  install.packages("readxl")
}

if(!require("ggplot2")){
  install.packages("ggplot2")
}

if(!require("caret")){
  install.packages("caret")
}

if(!require("glmnet")){
  install.packages("glmnet")
}

if(!require("WeightedROC")){
  install.packages("WeightedROC")
}

if(!require("gbm")){
  install.packages("gbm")
}

if(!require("xgboost")){
  install.packages("xgboost")
}

if(!require("tidyverse")){
  install.packages("tidyverse")
}

library(R.matlab)
library(readxl)
library(dplyr)
library(EBImage)
library(ggplot2)
library(caret)
library(glmnet)
library(WeightedROC)
library(gbm)
library(xgboost)
library(tidyverse)

Step 0 set work directories

set.seed(2020)
setwd(getwd())

Provide directories for training images. Training images and Training fiducial points will be in different subfolders.

train_dir <- "../data/train_set/" # This will be modified for different data sets.
train_image_dir <- paste(train_dir, "images/", sep="")
train_pt_dir <- paste(train_dir,  "points/", sep="")
train_label_path <- paste(train_dir, "label.csv", sep="") 

Step 1: set up controls for evaluation experiments.

In this chunk, we have a set of controls for the evaluation experiments.

• (T/F) cross-validation on the training set
• (T/F) reweighting the samples for training set
• (number) K, the number of CV folds
• (T/F) process features for training set
• (T/F) run evaluation on an independent test set
• (T/F) process features for test set

run.cv <- FALSE # run cross-validation on the training set
sample.reweight <- TRUE # run sample reweighting in model training
K <- 5  # number of CV folds
run.feature.train <- TRUE # process features for training set
run.test <- TRUE # run evaluation on an independent test set
run.feature.test <- TRUE # process features for test set

run.baseline <- TRUE # run baseline model
run.xgboost <- TRUE # run xgboost model

run.pred <- TRUE

Using cross-validation or independent test set evaluation, we compare the performance of models with different specifications. In this part, we tune parameters n.trees (total number of trees) and shrinkage (learning rate) for boosted decision stumps (Gradient Boosting Machine). And tune parameters eta (learning rate) and colsample_bytree (subsample ratio of columns when constructing each tree) for xgboost model (eXtreme Gradient Boosting Training).

# GBM parameters
n.trees  <- c(40,50,60,70) #40-70
shrinkage <- c(0.01,0.05,0.1,0.15) #0.01-1

#model_labels = paste("GBM with n.trees =", n.trees,"and shrinkage =",shrinkage) 

# xgboost parameters
e <-c(0.05, 0.1, 0.2, 0.3, 0.4, 0.5) # eta
c <- c(0.2,0.5,0.8,1) # colsample_bytree

Step 2: import data and train-test split

#train-test split
info <- read.csv(train_label_path)
n <- nrow(info)
n_train <- round(n*(4/5), 0)
train_idx <- sample(info$Index, n_train, replace = F)
test_idx <- setdiff(info$Index, train_idx)

If you choose to extract features from images, such as using Gabor filter, R memory will exhaust all images are read together. The solution is to repeat reading a smaller batch(e.g 100) and process them.

n_files <- length(list.files(train_image_dir))

image_list <- list()
for(i in 1:100){
   image_list[[i]] <- readImage(paste0(train_image_dir, sprintf("%04d", i), ".jpg"))
}

Fiducial points are stored in matlab format. In this step, we read them and store them in a list.

#function to read fiducial points
#input: index
#output: matrix of fiducial points corresponding to the index
readMat.matrix <- function(index){
     return(round(readMat(paste0(train_pt_dir, sprintf("%04d", index), ".mat"))[[1]],0))
}

#load fiducial points
fiducial_pt_list <- lapply(1:n_files, readMat.matrix)
save(fiducial_pt_list, file="../output/fiducial_pt_list.RData")

Step 3: construct features and responses

• The follow plots show how pairwise distance between fiducial points can work as feature for facial emotion recognition.

  • In the first column, 78 fiducials points of each emotion are marked in order.
  • In the second column distributions of vertical distance between right pupil(1) and right brow peak(21) are shown in histograms. For example, the distance of an angry face tends to be shorter than that of a surprised face.
  • The third column is the distributions of vertical distances between right mouth corner(50) and the midpoint of the upper lip(52). For example, the distance of an happy face tends to be shorter than that of a sad face.

Figure1

feature.R should be the wrapper for all your feature engineering functions and options. The function feature( ) should have options that correspond to different scenarios for your project and produces an R object that contains features and responses that are required by all the models you are going to evaluate later.

• feature.R
• Input: list of images or fiducial point
• Output: an RData file that contains extracted features and corresponding responses

source("../lib/feature.R")
tm_feature_train <- NA
if(run.feature.train){
  tm_feature_train <- system.time(dat_train <- feature(fiducial_pt_list, train_idx))
  save(dat_train, file="../output/feature_train.RData")
}else{
  load(file="../output/feature_train.RData")
}

tm_feature_test <- NA
if(run.feature.test){
  tm_feature_test <- system.time(dat_test <- feature(fiducial_pt_list, test_idx))
  save(dat_test, file="../output/feature_test.RData")
}else{
  load(file="../output/feature_test.RData")
}

Step 4: Train a classification model with training features and responses

Call the train model and test model from library.

train.R and test.R should be wrappers for all your model training steps and your classification/prediction steps.

• train.R
  • Input: a data frame containing features and labels and a parameter list.
  • Output:a trained model
• test.R
  • Input: the fitted classification model using training data and processed features from testing images
  • Input: an R object that contains a trained classifier.
  • Output: training model specification
• In this part, we use GBM (baseline model) and xgboost (advanced model) to do classification.

(a). GBM (baseline model)

source("../lib/train_baseline.R") 
source("../lib/test_baseline.R")
source("../lib/cv_baseline.R")

Model selection with cross-validation

• Do model selection by choosing among different values of training model parameters.

feature_train = as.matrix(dat_train[, -6007]) 
label_train = as.integer(dat_train$label)

if(run.cv){
  
mean.error_cv <- matrix(0, nrow = length(n.trees), ncol=length(shrinkage))
sd.error_cv <- matrix(0, nrow = length(n.trees), ncol=length(shrinkage))
mean.AUC_cv <- matrix(0, nrow = length(n.trees), ncol=length(shrinkage))
sd.AUC_cv <- matrix(0, nrow = length(n.trees), ncol=length(shrinkage))

  for(i in 1:length(n.trees)){
    cat("n.trees = ", n.trees[i], "\n")
    
    for (j in 1:length(shrinkage)){
    cat("shrinkage = ", shrinkage[j], "\n")
      
      res_cv<- cv_baseline(feature_train, 
                              label_train, 
                              K, 
                              n.trees[i],
                              shrinkage[j],
                           reweight = sample.reweight)
      
      mean.error_cv[i,j]<-res_cv[1]
      sd.error_cv[i,j]<-res_cv[2]
      mean.AUC_cv[i,j]<-res_cv[3]
      sd.AUC_cv[i,j]<-res_cv[4]
    }
  }
save(mean.error_cv, file="../output/mean.error_cv.RData")
save(sd.error_cv, file="../output/sd.error_cv.RData")
save(mean.AUC_cv, file="../output/mean.AUC_cv.RData")
save(sd.AUC_cv, file="../output/sd.AUC_cv.RData")
}else{
  load("../output/mean.error_cv.RData")
  load("../output/sd.error_cv.RData")
  load("../output/mean.AUC_cv.RData")
  load("../output/sd.AUC_cv.RData")
  }

df.mean.error<-data.frame(mean.error_cv)%>%
  setNames(shrinkage)%>%
  mutate(n.trees=n.trees)%>%
  pivot_longer(c(-n.trees),names_to = "shrinkage",values_to = "mean.error")

df.sd.error<-data.frame(sd.error_cv)%>%
  setNames(shrinkage)%>%
  mutate(n.trees=n.trees)%>%
  pivot_longer(c(-n.trees),names_to = "shrinkage",values_to = "sd.error")

df.mean.AUC<-data.frame(mean.AUC_cv)%>%
  setNames(shrinkage)%>%
  mutate(n.trees=n.trees)%>%
  pivot_longer(c(-n.trees),names_to = "shrinkage",values_to = "mean.AUC")

df.sd.AUC<-data.frame(sd.AUC_cv)%>%
  setNames(shrinkage)%>%
  mutate(n.trees=n.trees)%>%
  pivot_longer(c(-n.trees),names_to = "shrinkage",values_to = "sd.AUC")

res_cvvv<-df.mean.error%>%mutate(sd.error=df.sd.error$sd.error,
                                 mean.AUC=df.mean.AUC$mean.AUC,
                                 sd.AUC=df.sd.AUC$sd.AUC)

save(res_cvvv, file="../output/res_cvvv.RData")

Visualize cross-validation results.

load("../output/res_cvvv.RData")


  p1 <- res_cvvv%>%
    ggplot(aes(x = n.trees, y = mean.error,
               ymin = mean.error - sd.error, ymax = mean.error + sd.error)) + 
    geom_crossbar() +
    facet_wrap(~shrinkage)+
    theme(axis.text.x = element_text(angle = 90, hjust = 1))
  
  p2 <- res_cvvv %>% 
    ggplot(aes(x = as.factor(n.trees), y = mean.AUC,
               ymin = mean.AUC - sd.AUC, ymax = mean.AUC + sd.AUC)) + 
    geom_crossbar() +
    facet_wrap(~shrinkage)+
    theme(axis.text.x = element_text(angle = 90, hjust = 1))
  
  print(p1)
  print(p2)

• Choose the “best” parameter value

n.trees_best <- as.numeric(res_cvvv[which.min(res_cvvv$mean.error),1])
shrinkage_best <- as.numeric(res_cvvv[which.min(res_cvvv$mean.error),2])

n.trees_best
shrinkage_best

• Train the model with the entire training set using the selected model (model parameter) via cross-validation. The “best” parameters we got here are n.trees=50 and shrinkage=0.15.

# training weights
weight_train <- rep(NA, length(label_train))
for (v in unique(label_train)){
  weight_train[label_train == v] = 0.5 * length(label_train) / length(label_train[label_train == v])
}

if (sample.reweight){
  tm_train <- system.time(fit_train <- train_baseline(feature_train,label_train,
                                             w = weight_train,
                                             n.trees_best,shrinkage_best))
} else {
  tm_train <- system.time(fit_train <- train_baseline(feature_train,label_train,
                                             w=NULL,
                                             n.trees_best,shrinkage_best))
}
save(fit_train, file="../output/fit_train.RData")

(b). xgboost (advanced model)

source("../lib/train_xgboost.R") 
source("../lib/test_xgboost.R")
source("../lib/cv_xgboost.R")

Model selection with cross-validation

• Do model selection by choosing among different values of training model parameters.

feature_train = as.matrix(dat_train[, -6007])
label_train = as.integer(dat_train$label)

if(run.cv){
  
  mean.error_cv_xgb <- matrix(0, nrow = length(e), ncol=length(c))
  sd.error_cv_xgb <- matrix(0, nrow = length(e), ncol=length(c))
  mean.AUC_cv_xgb <- matrix(0, nrow = length(e), ncol=length(c))
  sd.AUC_cv_xgb <- matrix(0, nrow = length(e), ncol=length(c))

  for(i in 1:length(e)){
    for (j in 1:length(c)){
      
      cat("eta = ", e[i], ", colsample_bytree = ", c[j], "\n")
      
      res_cv_xgb <- cv_xgb(dat_train, K, para=list(eta=e[i], colsample_bytree=c[j], 
                                               max_depth=6, min_child_weight=1))
      mean.error_cv_xgb[i,j]<-res_cv_xgb[1]
      sd.error_cv_xgb[i,j]<-res_cv_xgb[2]
      mean.AUC_cv_xgb[i,j]<-res_cv_xgb[3]
      sd.AUC_cv_xgb[i,j]<-res_cv_xgb[4]
    }
  }
  
  save(mean.error_cv_xgb, file="../output/mean.error_cv_xgb.RData")
  save(sd.error_cv_xgb, file="../output/sd.error_cv_xgb.RData")
  save(mean.AUC_cv_xgb, file="../output/mean.AUC_cv_xgb.RData")
  save(sd.AUC_cv_xgb, file="../output/sd.AUC_cv_xgb.RData")

}else{
  load("../output/mean.error_cv_xgb.RData")
  load("../output/sd.error_cv_xgb.RData")
  load("../output/mean.AUC_cv_xgb.RData")
  load("../output/sd.AUC_cv_xgb.RData")
}

Visualize cross-validation results.

if(run.cv){
df.mean.error_xgb <- data.frame(mean.error_cv_xgb)%>%
  setNames(c)%>%
  mutate(eta=e)%>%
  pivot_longer(c(-eta),names_to = "colsample_bytree",values_to = "mean.error")

df.sd.error_xgb <- data.frame(sd.error_cv_xgb)%>%
  setNames(c)%>%
  mutate(eta=e)%>%
  pivot_longer(c(-eta),names_to = "colsample_bytree",values_to = "sd.error")

df.mean.AUC_xgb <- data.frame(mean.AUC_cv_xgb)%>%
  setNames(c)%>%
  mutate(eta=e)%>%
  pivot_longer(c(-eta),names_to = "colsample_bytree",values_to = "mean.AUC")

df.sd.AUC_xgb <- data.frame(sd.AUC_cv_xgb)%>%
  setNames(c)%>%
  mutate(eta=e)%>%
  pivot_longer(c(-eta),names_to = "colsample_bytree",values_to = "sd.AUC")

df.res_cv_xgb <- df.mean.error_xgb%>%mutate(sd.error=df.sd.error_xgb$sd.error,
                                 mean.AUC=df.mean.AUC_xgb$mean.AUC,
                                 sd.AUC=df.sd.AUC_xgb$sd.AUC)

save(df.res_cv_xgb, file="../output/res_cv_xgb.RData")
}else{
  load("../output/res_cv_xgb.RData")
}

  p1 <- df.res_cv_xgb%>%
    ggplot(aes(x = eta, y = mean.error,
               ymin = mean.error - sd.error, ymax = mean.error + sd.error)) + 
    geom_crossbar() +
    facet_wrap(~colsample_bytree)+
    theme(axis.text.x = element_text(angle = 90, hjust = 1))
  
  p2 <- df.res_cv_xgb %>% 
    ggplot(aes(x = as.factor(eta), y = mean.AUC,
               ymin = mean.AUC - sd.AUC, ymax = mean.AUC + sd.AUC)) + 
    geom_crossbar() +
    facet_wrap(~colsample_bytree)+
    theme(axis.text.x = element_text(angle = 90, hjust = 1))
  
  print(p1)
  print(p2)

• Choose the “best” parameter value

eta_best <- as.numeric(df.res_cv_xgb[which.min(df.res_cv_xgb$mean.error),1])
colsample_bytree_best <- as.numeric(df.res_cv_xgb[which.min(df.res_cv_xgb$mean.error),2])

eta_best
colsample_bytree_best

• Train the model with the entire training set using the selected model (model parameter) via cross-validation. The “best” parameters we got here are eta=0.05 and colsample_bytree=0.2.

para <- list(objective="binary:logistic", eta=eta_best, max_depth=6, min_child_weight=1,
            colsample_bytree=colsample_bytree_best)
if(run.xgboost){
  train_result_xgb <- xgb_train(dat_train, para)
  model_xgb <- train_result_xgb[[1]]
  train_time_xgb <- train_result_xgb[[2]]
  save(model_xgb, file="../output/model_xgb.RData")
  save(train_time_xgb, file="../output/train_time_xgb.RData")
}

Step 5: Run test on test images

(a). GBM (baseline model)

tm_test = NA
feature_test <- as.matrix(dat_test[, -6007])

if(run.test){
  load(file="../output/fit_train.RData")
  tm_test <- system.time(
    {label_pred <- as.integer(test_baseline(fit_train,feature_test,
                                   n.trees_best,shrinkage_best, 
                                   pred.type = 'link'));
    
    prob_pred <- test_baseline(fit_train, feature_test, 
                      n.trees_best,shrinkage_best,
                      pred.type = 'response')})
}

• evaluation

## reweight the test data to represent a balanced label distribution

label_test <- as.integer(dat_test$label)

weight_test <- rep(NA, length(label_test))

for (v in unique(label_test)){
  weight_test[label_test == v] = 0.5 * length(label_test) / length(label_test[label_test == v])
}

accu <- mean(label_pred == label_test)
tpr.fpr <- WeightedROC(prob_pred, label_test, weight_test)
auc <- WeightedAUC(tpr.fpr)

cat("The accuracy of model: GBM with n.trees =", n.trees_best, "and shrinkage =",shrinkage_best,"is", accu*100, "%.\n")
cat("The AUC of model: GBM with n.trees =", n.trees_best, "and shrinkage =",shrinkage_best,"is", auc, ".\n")

Summarize Running Time

Prediction performance matters, so does the running times for constructing features and for training the model, especially when the computation resource is limited.

cat("Time for constructing training features=", tm_feature_train[1], "s \n")
cat("Time for constructing testing features=", tm_feature_test[1], "s \n")
cat("Time for training model=", tm_train[1], "s \n") 
cat("Time for testing model=", tm_test[1], "s \n")

(b). xgboost (advanced model)

if(run.xgboost){
  load("../output/model_xgb.RData")
  test_result_xgb = xgb_test(model_xgb, dat_test)
  pred_xgb = test_result_xgb[[1]]
  pred_xgb = ifelse(pred_xgb>0.5, 1, 0)
  test_time_xgb = test_result_xgb[[2]]
}

• evaluation

## reweight the test data to represent a balanced label distribution

if(run.xgboost){
  
  accuracy_xgb = mean(dat_test$label == pred_xgb)
  
  label_test <- as.numeric(dat_test$label)-1
  weight_test <- rep(NA, length(label_test))
  for (v in unique(label_test)){
    weight_test[label_test == v] = 0.5 * length(label_test) / length(label_test[label_test == v])
  }

  tpr.fpr <- WeightedROC(pred_xgb, label_test, weight_test)
  auc <- WeightedAUC(tpr.fpr)
  
  cat("The accuracy of xgb model is:", accuracy_xgb*100, "%.\n")
  cat("The AUC of xgb model is:", auc, ".\n")
  
}

The accuracy of xgb model is: 86.66667 %.
The AUC of xgb model is: 0.6918698 .

Summarize Running Time

Prediction performance matters, so does the running times for constructing features and for training the model, especially when the computation resource is limited.

if(run.xgboost){
  cat("Time for constructing training features=", tm_feature_train[1], "s \n")
  cat("Time for constructing testing features=", tm_feature_test[1], "s \n")
  cat("Time for training model=", train_time_xgb[1], "s \n") 
  cat("Time for testing model=", test_time_xgb[1], "s \n")
}

Step 6: Prediction

if(run.pred){
pred_dir <- "../data/test_set_predict/" # This will be modified for different data sets.
pred_image_dir <- paste(pred_dir, "images/", sep="")
pred_pt_dir <- paste(pred_dir,  "points/", sep="")
pred_label_path <- paste(pred_dir, "label_prediction.csv", sep="")

pred_info <- read.csv(pred_label_path)
pred_n <- nrow(pred_info)
pred_idx <- pred_info$Index

pred_n_files <- length(list.files(pred_image_dir))

#load fiducial points
pred_fiducial_pt_list <- lapply(1:pred_n_files, readMat.matrix) 

save(pred_fiducial_pt_list, file="../output/pred_fiducial_pt_list.RData")

source("../lib/feature.R")

tm_feature_pred <- NA
if(run.feature.test){
  tm_feature_pred <- system.time(dat_pred <- feature(pred_fiducial_pt_list, pred_idx))
  save(dat_pred, file="../output/feature_pred.RData")
}else{
  load(file="../output/feature_pred.RData")
}

feature_pred <- as.matrix(dat_pred[, -6007])
}

cannot open file '../data/train_set/points/0001.mat': No such file or directoryError in file(con, open = "rb") : cannot open the connection

(a). GBM (baseline model)

tm_pred = NA

if(run.pred){
  load(file="../output/fit_train.RData")
  tm_pred <- system.time(
    {label_pred <- as.integer(test_baseline(fit_train,feature_pred,
                                   n.trees_best,shrinkage_best, 
                                   pred.type = 'link'))-1})
  pred_info<-pred_info%>%mutate(Baseline=label_pred)
}

Summarize Running Time

Prediction performance matters, so does the running times for constructing features and for training the model, especially when the computation resource is limited.

if(run.pred){
cat("Predicting time of baseline model=", tm_pred[1], "s \n")}

(b). xgboost (advanced model)

if(run.pred){
  load("../output/model_xgb.RData")
  pred_result_xgb = xgb_test(model_xgb, dat_pred)
  final_pred_xgb = pred_result_xgb[[1]]
  final_pred_xgb = ifelse(final_pred_xgb>0.5, 1, 0)
  pred_time_xgb = pred_result_xgb[[2]]
  pred_info<-pred_info%>%mutate(Advanced=final_pred_xgb)
}

Summarize Running Time

Prediction performance matters, so does the running times for constructing features and for training the model, especially when the computation resource is limited.

if(run.pred){
cat("Predicting time of advanced model=", pred_time_xgb[1], "s \n")}

if(run.pred){write.csv(pred_info, file="../doc/label_prediction.csv")}

###Reference - Du, S., Tao, Y., & Martinez, A. M. (2014). Compound facial expressions of emotion. Proceedings of the National Academy of Sciences, 111(15), E1454-E1462.